package cn.com.w.bootfreemarker.util;

import cn.com.w.bootfreemarker.vo.FreemarkerDataModel;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;

/**
 * 使用步骤：
 * 第一步：创建一个 Configuration 对象，直接 new 一个对象。构造方法的参数就是 freemarker的版本号。
 * 第二步：设置模板文件所在的路径。
 * 第三步：设置模板文件使用的字符集。一般就是 utf-8.
 * 第四步：加载一个模板，创建一个模板对象。
 * 第五步：创建一个模板使用的数据集，可以是 pojo 也可以是 map。一般是 Map。
 * 第六步：创建一个 Writer 对象，一般创建一 FileWriter 对象，指定生成的文件名。
 * 第七步：调用模板对象的 process 方法输出文件。
 * 第八步：关闭流
 */
public class FreemarkerUtil {

    private static final Logger log = LoggerFactory.getLogger(FreemarkerUtil.class);

    private static Configuration configuration;

    static {
        // 1.创建freemarker.template.Configuration配置类
        configuration = new Configuration(Configuration.VERSION_2_3_23);
    }

    /**
     * 构建doc文档
     *
     * @param freemarkerDataModel 数据模型
     * @param ftlName             模板名称+后缀
     * @param filePath            doc文档生成绝对路径
     * @throws UnsupportedEncodingException
     */
    public static void buildDoc(FreemarkerDataModel freemarkerDataModel, String ftlName, String filePath)
            throws UnsupportedEncodingException {
        // 创建一个模板对象
        Template t = null;
        try {
            // 2.设置模板加载路径(所在目录)
            configuration.setDirectoryForTemplateLoading(new File(Thread.currentThread().getContextClassLoader().getResource("").getPath() + File.separator + "ftl"));
            // 3.设置字符集
            configuration.setDefaultEncoding("utf-8");
            // 4.加载模板，获取创建的XXX.ftl模板文件
            t = configuration.getTemplate(ftlName);
        } catch (IOException e) {
            log.error("读取{}模板文件失败！", ftlName, e);
        }
        // 定义输出的文件
        File outFile = new File(filePath);
        if (!outFile.getParentFile().exists()) {
            outFile.getParentFile().mkdirs();
        }
        // 创建输出流
        Writer out = null;
        // 基础流
        FileOutputStream fos = null;
        try {
            fos = new FileOutputStream(outFile);
            OutputStreamWriter oWriter = new OutputStreamWriter(fos, "UTF-8");
            out = new BufferedWriter(oWriter);
        } catch (FileNotFoundException e1) {
            e1.printStackTrace();
        }

        try {
            //  Executes template
            t.process(freemarkerDataModel, out);
            out.close();
            fos.close();
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}
